from classeCliente import *
from Repositorio_Cliente import *
from classeCarro import *
from Repositorio_Carro import *
from Locadora import *
#CLASSE CRIADA POR EDUARDO JOSE
op=1

while(op==1):
    #try:
        print("1 - MENU CLIENTE")
        print("2 - MENU CARRO")
        print("3 - LOCACAO")
        print("4 - DEVOLUCAO")
    
        op2=int(input("Entre com a opcao que deseja: \n"))
        if(op2==1):
            print ("MENU CLIENTE")
            print("1 - CADASTRO DE CLIENTE")
            print("2 - CONSULTA DE CLIENTE")
            print("3 - ALTERACAO DE CLIENTE")
            print("4 - EXCLUSAO DE CLIENTE")
            op3=int(input("Opcao desejada: \n"))
            if(op3==1):
                print '---CADASTRAR CLIENTE---'
                cpf = raw_input('CPF:\n')
                nome = raw_input('NOME:\n').upper()
                endereco = raw_input('ENDERECO:\n').upper()
                cliente = Cliente(nome,endereco,cpf)
                repositorio_cliente = Repositorio_Cliente()
                repositorio_cliente.cadastrar(cliente)
                
                

            if(op3==2):#OK!
                print'---CONSULTA DE CLIENTE---'
                cliente = Cliente()
                repositorio_cliente = Repositorio_Cliente()
                repositorio_cliente.pesquisar()

            if(op3==3):
                print '---ALTERAR DADOS DE CLIENTE---'
                print '---ESCOLHA O QUE DESEJA ALTERAR---'
                print '1 - NOME\n2 - ENDERECO\n3 - CPF\n'
                op4 = int(input('OPCAO:\n'))
                if(op4==1):
                    print '---ATUALIZACAO DE NOME---'
                    repositorio_cliente = Repositorio_Cliente()
                    repositorio_cliente.editarnome()

                if(op4==3):
                    print '---ATUALIZACAO DE CPF---'                    
                    repositorio_cliente = Repositorio_Cliente()
                    repositorio_cliente.editarcpf()

                if(op4==2):
                    print '---ATUALIZACAO DE ENDERECO---'                    
                    repositorio_cliente = Repositorio_Cliente()
                    repositorio_cliente.editarendereco()

            if(op3==4):
                print '---REMOVER CLIENTE---'                
                repositorio_cliente = Repositorio_Cliente()
                repositorio_cliente.remover()
        elif(op2 == 2):
            print("MENU CARRO") 
            print("1 - CADASTRO DE CARRO")
            print("2 - CONSULTA DE CARRO")
            print("3 - ALTERACAO DE CARRO")
            print("4 - EXCLUSAO DE CARRO")
            op5 = int(input("DIGITE A OPCAO DESEJADA: \n"))
            if(op5 == 1):
                print '---CADASTRAR CARRO---' 
                marca = raw_input("MARCA:\n").upper()
                modelo = raw_input("MODELO:\n").upper()
                cor = raw_input("COR:\n").upper()
                ano = raw_input("ANO:\n").upper()
                placa = raw_input("PLACA:\n")
                carro = Carro(marca,modelo,ano,cor,placa)
                repositorio_carro = Repositorio_Carro()
                repositorio_carro.Cadastrar_Carro(carro)
                print ("CARRO CADASTRADO COM SUCESSO!")

            if(op5 == 2):
                print '---CONSULTA DE CARRO---'
                
                repositorio_carro = Repositorio_Carro()
                repositorio_carro.pesquisarCarro()
                    
            elif(op5 == 3):
                print '---ALTERACAO DE CARRO---'
                repositorio_carro = Repositorio_Carro()
                placa = raw_input("PLACA:")
                carro = repositorio_carro.pesquisarCarro2(placa)
                print '---ESCOLHA O QUE DESEJA ALTERAR---'
                print '1 - COR\n2 - MODELO\n3 - MARCA\n4 - ANO'
                opAlterar = int(input("OPCAO:\n"))
                if(opAlterar == 1):
                    print '---ATUALIZACAO DE COR---'
                    novaCor = raw_input("NOVA COR:")
                    carro.setCor(novaCor)
                    repositorio_carro.refreshDb(carro)

                if(opAlterar == 2):
                    print '---ATUALIZACAO DE MODELO---'
                    novoModelo = raw_input("NOVO MODELO:")
                    carro.setModelo(novoModelo)
                    repositorio_carro.refreshDb(carro)
                if(opAlterar == 3):
                    print '---ATUALIZACAO DE MARCA---'
                    novaMarca = raw_input("NOVA MARCA:")
                    carro.setMarca(novaMarca)
                    repositorio_carro.refreshDb(carro)
                if(opAlterar == 4):
                    print '---ATUALIZACAO DE ANO---'
                    novoAno = raw_input("NOVO ANO:")
                    carro.setAno(novoAno)
                    repositorio_carro.refreshDb(carro)

            if(op5 == 4):
                print '---EXCLUSAO DE CARRO---'
                repositorio_carro = Repositorio_Carro()
                placa = raw_input("PLACA:")
                carro = repositorio_carro.pesquisarCarro2(placa)
                repositorio_carro.excluirCarro(carro)
                

        if(op2==3):
            print '---LOCACAO---'
            locadora = Locadora()
            locadora.locar()

        if(op2 == 4):
            print '---DEVOLUCAO---'
            locadora = Locadora()
            locadora.devolver()

   # except:
       # print 'OPCAO INVALIDA'
